﻿//3128. 直角三角形
//给你一个二维 boolean 矩阵 grid 。
//如果 grid 的 3 个元素的集合中，一个元素与另一个元素在 同一行，并且与第三个元素在 同一列，则该集合是一个 直角三角形。3 个元素 不必 彼此相邻。
//请你返回使用 grid 中的 3 个元素可以构建的 直角三角形 数目，且满足 3 个元素值 都 为 1 。


class Solution {
public:
    long long numberOfRightTriangles(vector<vector<int>>& grid)
    {
        int m = grid.size();
        int n = grid[0].size();
        vector<int> row(m, 0);
        vector<int> col(n, 0);
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (grid[i][j])
                {
                    row[i]++;
                    col[j]++;
                }
            }
        }
        long long ans = 0;
        for (int i = 0; i < m; i++)
        {
            for (int j = 0; j < n; j++)
            {
                if (grid[i][j] && row[i] && col[j])
                {
                    ans = ans + (row[i] - 1) * (col[j] - 1);
                }
            }
        }
        return ans;
    }
};